from .vq_model import VQGAN